"""
复现reference中的研报并加以改进，数据使用股指期货IC、IF的日内数据
"""
import pandas as pd

# 设置value的显示长度为200，默认为50
pd.set_option('max_colwidth', 200)
# 显示所有列，把行显示设置成最大
pd.set_option('display.max_columns', None)
# 显示所有行，把列显示设置成最大
pd.set_option('display.max_rows', None)

# 打开HDF5文件
with pd.HDFStore("dataset/IF.h5", 'r') as store:
    # 获取文件中所有的日期（Group）
    dates = list(store.keys())
    print(len(dates))
    print(dates[0])
    print(dates[-1])
    data = store.get(dates[0])
    # print(dates[:5])
    print(data.info(verbose=True))
    # print(data.columns)
    # print(data.loc[:,'datetime'].unique().tolist())
    print(data.iloc[200])
    # # 循环遍历每个日期（Group），读取对应的DataFrame
    # for date in dates[:1]:
    #     # 通过日期（Group）名称来获取DataFrame
    #     data = store.get(date)
    #     print("Data for date:", date)
    #     print(data.head())
